#ifndef SELFATTENTIONBINDER_H 
#define SELFATTENTIONBINDER_H 
#include "hosttensor_binder.h" 
#include <vector> 
namespace atb_speed { 
class SelfAttentionBinder : public HostTensorBinder { 
public: 
    SelfAttentionBinder(); 
    virtual ~SelfAttentionBinder(); 
    void ParseParam(const nlohmann::json &paramJson) override; 
    void BindTensor(atb::VariantPack &variantPack) override; 
    
private: 
    std::vector<int32_t> tokenOffset_; 
    std::vector<int32_t> seqLen_; 
}; 
} 
#endif